#!/bin/bash

# 修改权限：chmod +x ./test.sh

if [ -n "$1" ] #存在参数
then
    epoch=$1
    echo ">>epoch is $epoch"
else
    echo "没有epoch参数，比如:./train.sh 300"
    exit 8
fi


iresnet50
# align
#python train.py --name transFER_mutlFeatureAgg --arch=transFER_mutlFeatureAgg --epochs=10 --opti=SGD --num_workers=2 --lr=0.00025 --batch_size=32 --mixup=True --label_smooth=True --data_type=fer2013 --data_path=D:\WorkSpace\TransMultiScaleFusion\dataset\fer2013.csv
#python train.py --name CK+_Basic_IR_MFE_MDS_Att_MFA --arch=Basic_IR_MFE_MDS_Att_MFA --epochs=10 --opti=SGD --num_workers=2 --lr=0.00012 --batch_size=16 --mixup=True --label_smooth=True  --data_type=ck+ -data_path=
#python train.py --name RAFDB_Basic_IR_MFE_MDS_Att_MFA --arch=Basic_IR_MFE_MDS_Att_MFA --epochs=10 --opti=SGD --num_workers=2 --lr=0.00012 --batch_size=16 --mixup=True --label_smooth=True  --data_type=rafdb -data_path=

python train.py --name iresnet50_pretrain --arch=iresnet50 --epochs=200 --opti=Adam --num_workers=2 --lr=0.01 --batch_size=64 --mixup=True --label_smooth=True --data_type=fer2013 --data_path=D:\WorkSpace\TransMultiScaleFusion\dataset\fer2013.csv

